def zs1(fun):
    def inner(*args,**kwargs):
      print("begin")
      ret = fun(*args,**kwargs)
      print("end")
      return ret
    return inner

@zs1
def test():
    print("test")

test()

@zs1
def test2(a,b):
    print("{}+{}={}".format(a,b,a+b))

test2(3,5)